import hashlib
import json
import aiofiles
from os.path import isfile

from pddnsc.base import BaseFilterProvider


class GenericTextFileFilter(BaseFilterProvider):
    """Проверка на то что хотябы один IP адрес изменился по сравнению с текстом в файле

    Конфигурация:

      - filepath: имя файла
      - encoding: кодировка, по умолчанию "utf-8"
      - mode: режим открытия, по умолчанию "r"
      - check_ipv4: проверять ли IPv4, по умолчанию нет если check_ipv6 есть в конфиге иначе да
      - check_ipv6: проверять ли IPv6, по умолчанию нет если check_ipv4 есть в конфиге иначе да
    """

    def post_init(self):
        super().post_init()
        self.filepath = self.config["filepath"]
        self.encoding = self.config.get("encoding", "utf-8")
        self.mode = self.config.get("mode", "r")
        self.check_ipv4 = self.config.get("check_ipv4", False)
        self.check_ipv6 = self.config.get("check_ipv6", False)
        if "check_ipv4" not in self.config and "check_ipv4" not in self.config:
            self.check_ipv4 = self.check_ipv6 = True
        self.content = ""

    async def read(self) -> str:
        async with aiofiles.open(
            self.filepath, mode=self.mode, encoding=self.encoding
        ) as f:
            self.content = await f.read()

    async def check_imp(self, source_provider: str, addr_v4: str, addr_v6: str) -> bool:
        lst = []
        if self.check_ipv4:
            lst.append(addr_v4)
        if self.check_ipv4:
            lst.append(addr_v6)
        new_content = "\n".join(lst)
        return new_content != self.content

    async def check(self, source_provider: str, addr_v4: str, addr_v6: str) -> bool:
        if not isfile(self.filepath):
            return True
        await self.read()
        return await self.check_imp(source_provider, addr_v4, addr_v6)


class StateHashFilter(GenericTextFileFilter):
    """Проверка на то что хотябы один IP адрес изменился по хешу сохраненному в файле"""

    async def check_imp(self, source_provider: str, addr_v4: str, addr_v6: str) -> bool:
        new_state_str = (self.check_ipv4 and addr_v4 or "") + (
            self.check_ipv6 and addr_v6 or ""
        )
        new_sha = hashlib.sha256(new_state_str.encode(encoding=self.encoding))
        return self.content != new_sha.hexdigest()


class StateFileFilter(GenericTextFileFilter):
    """Проверка на то что хотябы один IP адрес изменился по сравнению с данными в json файле"""

    async def check_imp(self, source_provider: str, addr_v4: str, addr_v6: str) -> bool:
        new_state = {}
        if self.check_ipv4:
            new_state["ipv4"] = addr_v4 or ""
        if self.check_ipv6:
            new_state["ipv6"] = addr_v6 or ""

        old_state = json.loads(self.content)
        result = True

        if self.check_ipv4:
            result = result and new_state["ipv4"] != old_state["ipv4"]

        if self.check_ipv6:
            result = result and new_state["ipv6"] != old_state["ipv6"]

        return result
